FROM python:3.9.23

# 设置工作目录
WORKDIR /app

# 安装项目所需依赖
COPY requirements.txt ./
# 指定torch
RUN pip install --no-cache-dir torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121
# 安装其它依赖
RUN pip install --no-cache-dir -r requirements.txt

# 将当前目录文件直接拷贝进容器(可选)
COPY . /app

# 暴露端口
EXPOSE 5000

# 启动命令（使用 gunicorn+gevent）
CMD ["gunicorn", "-k", "gevent", "-b", "0.0.0.0:5000", "app:app"]
